AI-Assisted Plotting in Python